home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2004 #2 / K-CD-2-2004.ISO / OpenOffice Sv / f_0397 / python-core-2.2.2 / lib / test / test_pyexpat.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  6.4 KB  |  203 lines
  1. # Very simple test - Parse a file and print what happens
  2.  
  3. # XXX TypeErrors on calling handlers, or on bad return values from a
  4. # handler, are obscure and unhelpful.
  5.  
  6. import pyexpat
  7. from xml.parsers import expat
  8.  
  9. from test_support import sortdict
  10.  
  11. class Outputter:
  12.     def StartElementHandler(self, name, attrs):
  13.         print 'Start element:\n\t', repr(name), sortdict(attrs)
  14.  
  15.     def EndElementHandler(self, name):
  16.         print 'End element:\n\t', repr(name)
  17.  
  18.     def CharacterDataHandler(self, data):
  19.         data = data.strip()
  20.         if data:
  21.             print 'Character data:'
  22.             print '\t', repr(data)
  23.  
  24.     def ProcessingInstructionHandler(self, target, data):
  25.         print 'PI:\n\t', repr(target), repr(data)
  26.  
  27.     def StartNamespaceDeclHandler(self, prefix, uri):
  28.         print 'NS decl:\n\t', repr(prefix), repr(uri)
  29.  
  30.     def EndNamespaceDeclHandler(self, prefix):
  31.         print 'End of NS decl:\n\t', repr(prefix)
  32.  
  33.     def StartCdataSectionHandler(self):
  34.         print 'Start of CDATA section'
  35.  
  36.     def EndCdataSectionHandler(self):
  37.         print 'End of CDATA section'
  38.  
  39.     def CommentHandler(self, text):
  40.         print 'Comment:\n\t', repr(text)
  41.  
  42.     def NotationDeclHandler(self, *args):
  43.         name, base, sysid, pubid = args
  44.         print 'Notation declared:', args
  45.  
  46.     def UnparsedEntityDeclHandler(self, *args):
  47.         entityName, base, systemId, publicId, notationName = args
  48.         print 'Unparsed entity decl:\n\t', args
  49.  
  50.     def NotStandaloneHandler(self, userData):
  51.         print 'Not standalone'
  52.         return 1
  53.  
  54.     def ExternalEntityRefHandler(self, *args):
  55.         context, base, sysId, pubId = args
  56.         print 'External entity ref:', args[1:]
  57.         return 1
  58.  
  59.     def DefaultHandler(self, userData):
  60.         pass
  61.  
  62.     def DefaultHandlerExpand(self, userData):
  63.         pass
  64.  
  65.  
  66. def confirm(ok):
  67.     if ok:
  68.         print "OK."
  69.     else:
  70.         print "Not OK."
  71.  
  72. out = Outputter()
  73. parser = expat.ParserCreate(namespace_separator='!')
  74.  
  75. # Test getting/setting returns_unicode
  76. parser.returns_unicode = 0; confirm(parser.returns_unicode == 0)
  77. parser.returns_unicode = 1; confirm(parser.returns_unicode == 1)
  78. parser.returns_unicode = 2; confirm(parser.returns_unicode == 1)
  79. parser.returns_unicode = 0; confirm(parser.returns_unicode == 0)
  80.  
  81. # Test getting/setting ordered_attributes
  82. parser.ordered_attributes = 0; confirm(parser.ordered_attributes == 0)
  83. parser.ordered_attributes = 1; confirm(parser.ordered_attributes == 1)
  84. parser.ordered_attributes = 2; confirm(parser.ordered_attributes == 1)
  85. parser.ordered_attributes = 0; confirm(parser.ordered_attributes == 0)
  86.  
  87. # Test getting/setting specified_attributes
  88. parser.specified_attributes = 0; confirm(parser.specified_attributes == 0)
  89. parser.specified_attributes = 1; confirm(parser.specified_attributes == 1)
  90. parser.specified_attributes = 2; confirm(parser.specified_attributes == 1)
  91. parser.specified_attributes = 0; confirm(parser.specified_attributes == 0)
  92.  
  93. HANDLER_NAMES = [
  94.     'StartElementHandler', 'EndElementHandler',
  95.     'CharacterDataHandler', 'ProcessingInstructionHandler',
  96.     'UnparsedEntityDeclHandler', 'NotationDeclHandler',
  97.     'StartNamespaceDeclHandler', 'EndNamespaceDeclHandler',
  98.     'CommentHandler', 'StartCdataSectionHandler',
  99.     'EndCdataSectionHandler',
  100.     'DefaultHandler', 'DefaultHandlerExpand',
  101.     #'NotStandaloneHandler',
  102.     'ExternalEntityRefHandler'
  103.     ]
  104. for name in HANDLER_NAMES:
  105.     setattr(parser, name, getattr(out, name))
  106.  
  107. data = '''\
  108. <?xml version="1.0" encoding="iso-8859-1" standalone="no"?>
  109. <?xml-stylesheet href="stylesheet.css"?>
  110. <!-- comment data -->
  111. <!DOCTYPE quotations SYSTEM "quotations.dtd" [
  112. <!ELEMENT root ANY>
  113. <!NOTATION notation SYSTEM "notation.jpeg">
  114. <!ENTITY acirc "â">
  115. <!ENTITY external_entity SYSTEM "entity.file">
  116. <!ENTITY unparsed_entity SYSTEM "entity.file" NDATA notation>
  117. %unparsed_entity;
  118. ]>
  119.  
  120. <root attr1="value1" attr2="value2ὀ">
  121. <myns:subelement xmlns:myns="http://www.python.org/namespace">
  122.      Contents of subelements
  123. </myns:subelement>
  124. <sub2><![CDATA[contents of CDATA section]]></sub2>
  125. &external_entity;
  126. </root>
  127. '''
  128.  
  129. # Produce UTF-8 output
  130. parser.returns_unicode = 0
  131. try:
  132.     parser.Parse(data, 1)
  133. except expat.error:
  134.     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
  135.     print '** Line', parser.ErrorLineNumber
  136.     print '** Column', parser.ErrorColumnNumber
  137.     print '** Byte', parser.ErrorByteIndex
  138.  
  139. # Try the parse again, this time producing Unicode output
  140. parser = expat.ParserCreate(namespace_separator='!')
  141. parser.returns_unicode = 1
  142.  
  143. for name in HANDLER_NAMES:
  144.     setattr(parser, name, getattr(out, name))
  145. try:
  146.     parser.Parse(data, 1)
  147. except expat.error:
  148.     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
  149.     print '** Line', parser.ErrorLineNumber
  150.     print '** Column', parser.ErrorColumnNumber
  151.     print '** Byte', parser.ErrorByteIndex
  152.  
  153. # Try parsing a file
  154. parser = expat.ParserCreate(namespace_separator='!')
  155. parser.returns_unicode = 1
  156.  
  157. for name in HANDLER_NAMES:
  158.     setattr(parser, name, getattr(out, name))
  159. import StringIO
  160. file = StringIO.StringIO(data)
  161. try:
  162.     parser.ParseFile(file)
  163. except expat.error:
  164.     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
  165.     print '** Line', parser.ErrorLineNumber
  166.     print '** Column', parser.ErrorColumnNumber
  167.     print '** Byte', parser.ErrorByteIndex
  168.  
  169.  
  170. # Tests that make sure we get errors when the namespace_separator value
  171. # is illegal, and that we don't for good values:
  172. print
  173. print "Testing constructor for proper handling of namespace_separator values:"
  174. expat.ParserCreate()
  175. expat.ParserCreate(namespace_separator=None)
  176. expat.ParserCreate(namespace_separator=' ')
  177. print "Legal values tested o.k."
  178. try:
  179.     expat.ParserCreate(namespace_separator=42)
  180. except TypeError, e:
  181.     print "Caught expected TypeError:"
  182.     print e
  183. else:
  184.     print "Failed to catch expected TypeError."
  185.  
  186. try:
  187.     expat.ParserCreate(namespace_separator='too long')
  188. except ValueError, e:
  189.     print "Caught expected ValueError:"
  190.     print e
  191. else:
  192.     print "Failed to catch expected ValueError."
  193.  
  194. # ParserCreate() needs to accept a namespace_separator of zero length
  195. # to satisfy the requirements of RDF applications that are required
  196. # to simply glue together the namespace URI and the localname.  Though
  197. # considered a wart of the RDF specifications, it needs to be supported.
  198. #
  199. # See XML-SIG mailing list thread starting with
  200. # http://mail.python.org/pipermail/xml-sig/2001-April/005202.html
  201. #
  202. expat.ParserCreate(namespace_separator='') # too short
  203.